from enum import Enum


class StockStatusEnum(Enum):
    # 股票状态枚举
    HOLD = ("hold", "持有")
    FOLLOW = ("follow", "关注")
    STORAGE = ("storage", "入库")

    def __init__(self, key: str, description: str):
        self.key = key
        self.description = description

    def __str__(self):
        return self.key

    @classmethod
    def get_status_list(cls):
        """获取股票状态描述列表"""
        return [status.description for status in cls]

    @classmethod
    def get_default(cls):
        """获取默认状态"""
        return cls.HOLD.description
